# 这是一个示例 Python 脚本。

# 按 Shift+F10 执行或将其替换为您的代码。
# 按 双击 Shift 在所有地方搜索类、文件、工具窗口、操作和设置。


def print_hi(name):
    # 在下面的代码行中使用断点来调试脚本。
    print(f'Hi, {name}')  # 按 Ctrl+F8 切换断点。


# 按装订区域中的绿色按钮以运行脚本。
if __name__ == '__main__':
    print_hi('PyCharm')

# 访问 https://www.jetbrains.com/help/pycharm/ 获取 PyCharm 帮助
from segmentation import color_segmentation

if __name__ == "__main__":
    # 图像路径，因为img.png和main.py同目录，直接写文件名
    image_path = "img.png"
    # 示例：分割红色物体的HSV范围（可根据实际需求调整）
    lower_red = [0, 100, 100]
    upper_red = [10, 255, 255]

    color_segmentation(
        image_path=image_path,
        lower_bound=lower_red,
        upper_bound=upper_red,
        output_path="red_segmentation_result.png"
    )
